package com.leetcode.partition3;

import com.leetcode.common.TreeNode;

/**
 * @author `RKC`
 * @date 2021/8/21 9:54
 */
public class LC235二叉搜索树的最近公共祖先 {

    public static TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        return preorder(root, p, q);
    }

    private static TreeNode preorder(TreeNode root, TreeNode p, TreeNode q) {
        //BST树指定了搜索的方向，因此只需要按照BST的顺序性质前序搜索即可
        if (root == null) return null;
        //因为是BST树，如果root.val在[p.val, q.val]区间右边，则应该向左递归
        if (p.val < root.val && q.val < root.val) {
            return preorder(root.left, p, q);
        }
        if (p.val > root.val && q.val > root.val) {
            return preorder(root.right, p, q);
        }
        //前序遍历，因此第一个在p和q区间的节点就是最近公共祖先
        return root;
    }

    private static TreeNode iteration(TreeNode root, TreeNode p, TreeNode q) {
        while (root != null) {
            if (root.val < p.val && root.val < q.val) root = root.right;
            else if (root.val > p.val && root.val > q.val) root = root.left;
            else return root;
        }
        return null;
    }
}
